PFA: Attention Can Be Progressively Focused in Long-Sequence Generative Recommendation
PFA:长序列生成式推荐中注意力可渐进聚焦
🧭 生成式推荐 · 稀疏注意力加速 ★ 深度解读
Alibaba International Digital Commerce Group · Zhejiang University
利用 GR 解码中注意力渐进聚焦与跨步排名一致性,PFA 通过共享重要性排名和几何预算缩减,加速长序列生成式推荐
📌 工作参考价值(4 点)
- 对长序列生成式推荐,可借鉴"早期构造共享重要性排名 + 后续几何缩减预算",替代每步每 beam 独立 top-k,消除重复选择和 KV gather 开销,尤其适合 beam width 大、序列长的电商召回/广告候选生成。
- 工程实现上,用 beam 生成概率加权聚合注意力分布来挑选关键历史 token,比每个 beam 独立选更稳定且能共享连续 KV cache;可先 reorder KV cache,后续直接前缀截断,降低非连续内存访问。
- Step-aware KV projection:为不同解码步分配独立 KV 投影参数,成本低但能显著缓解复用 token 带来的表示退化,适合层级 SID 生成。
- 若业务采用生成式推荐,可保留 Step 0 全注意力做全局探索,不用过早压缩历史,对长序列效果稳定。